package com.ztom.dp;

/**
 * 打家劫舍
 * <p>
 * https://leetcode-cn.com/problems/house-robber/
 *
 * @author ZhangTao
 */
public class Code05Rob {

    public int rob2(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        if (nums.length == 1) {
            return nums[0];
        }

        return process(nums, nums.length - 1);
    }

    private int process(int[] nums, int i) {
        if (i == 0) {
            // 以 0 位置结尾, 只有一家
            return nums[0];
        }
        if (i == 1) {
            // 以 1 位置结尾, 只有两家
            return Math.max(nums[0], nums[1]);
        }

        return Math.max(
                // 以 i 位置结尾, 不偷 i 位置
                process(nums, i - 1),
                // 偷 i 位置
                process(nums, i - 2) + nums[i]
        );
    }

    public int rob1(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        int n = nums.length;
        if (n == 1) {
            return nums[0];
        }

        int[] dp = new int[n];
        dp[0] = nums[0];
        dp[1] = Math.max(nums[0], nums[1]);
        for (int i = 2; i < n; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + nums[i]);
        }

        return dp[n - 1];
    }

    public int rob(int[] nums) {
        if (nums == null || nums.length == 0) {
            return 0;
        }

        int n = nums.length;
        if (n == 1) {
            return nums[0];
        }

        int prepre = nums[0];
        int pre = Math.max(nums[0], nums[1]);
        int res = pre;
        for (int i = 2; i < n; i++) {
            res = Math.max(pre, prepre + nums[i]);
            prepre = pre;
            pre = res;
        }

        return res;
    }
}
